File Handling in Python Assignment